package com.mnsk.generator.domain;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.mnsk.generator.common.cache.GenCacheUtils;
import com.mnsk.generator.common.constant.GenConstants;
import com.mnsk.generator.common.core.domain.BaseEntity;
import com.mnsk.generator.common.util.GenUtils;
import com.mnsk.generator.common.utils.StringUtils;
import com.mnsk.generator.common.validator.StringSeries;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;

import javax.validation.Valid;
import javax.validation.constraints.NotBlank;
import java.util.List;

/**
 * 代码生成业务表
 *
 * @author admin
 */
@Data
@Accessors(chain = true)
@TableName("t_table")
@EqualsAndHashCode(callSuper = false)
public class Table extends BaseEntity {

    /**
     * 主键
     */
    @TableId(type = IdType.AUTO)
    private Long tableId;
    /**
     * 数据源主键
     */
    private Long dataSourceId;
    /**
     * 表名称
     */
    @NotBlank(message = "表名称不能为空")
    private String tableName;
    /**
     * 表描述
     */
    @NotBlank(message = "表描述不能为空")
    private String tableComment;
    /**
     * 关联父表的表名
     */
    private String subTableName;
    /**
     * 本表关联父表的外键名
     */
    private String subTableFkName;
    /**
     * 实体类名称(首字母大写)
     */
    @NotBlank(message = "实体类名称不能为空")
    private String className;
    /**
     * 生成操作模板（crud单表操作 tree树表操作 sub主子表操作）
     */
    private String tplCategory;
    /**
     * 生成包路径
     */
    @NotBlank(message = "生成包路径不能为空")
    private String packageName;
    /**
     * 生成模块名
     */
    @NotBlank(message = "生成模块名不能为空")
    private String moduleName;
    /**
     * 生成业务名
     */
    @NotBlank(message = "生成业务名不能为空")
    private String businessName;
    /**
     * 生成功能名
     */
    @NotBlank(message = "生成功能名不能为空")
    private String functionName;
    /**
     * 生成作者
     */
    @NotBlank(message = "作者不能为空")
    private String functionAuthor;
    /**
     * 生成代码方式（0 zip压缩包 1 自定义路径）
     */
    @StringSeries(list = {"0"})
    private String genType;
    /**
     * 生成路径（不填默认项目路径）
     */
    private String genPath;
    /**
     * 其它生成选项
     */
    @TableField("`options`")
    private String options;

    /**
     * 数据源类型
     */
    @TableField(exist = false)
    private String dbType;

    /**
     * 树编码字段
     */
    @TableField(exist = false)
    private String treeCode;
    /**
     * 树父编码字段
     */
    @TableField(exist = false)
    private String treeParentCode;
    /**
     * 树名称字段
     */
    @TableField(exist = false)
    private String treeName;
    /**
     * 上级菜单ID字段
     */
    @TableField(exist = false)
    private String parentMenuId;
    /**
     * 上级菜单名称字段
     */
    @TableField(exist = false)
    private String parentMenuName;

    /**
     * 主键信息
     */
    @TableField(exist = false)
    private TableColumn pkColumn;
    /**
     * 子表信息
     */
    @TableField(exist = false)
    private Table subTable;

    /**
     * 表列信息
     */
    @Valid
    @TableField(exist = false)
    private List<TableColumn> columns;

    public String getPackageName() {
        return GenCacheUtils.getPackageName();
    }

    public String getFunctionAuthor() {
        return GenCacheUtils.getAuthor();
    }

    public String getModuleName() {
        return GenUtils.getModuleName(GenCacheUtils.getPackageName());
    }

    public boolean isSub() {
        return isSub(this.tplCategory);
    }

    public boolean containColumn(String columnName) {
        for (TableColumn tableColumn : columns) {
            if (tableColumn.getColumnName().equals(columnName)) {
                return true;
            }
        }
        return false;
    }

    public boolean likeColumn(String columnName) {
        for (TableColumn tableColumn : columns) {
            if (tableColumn.getColumnName().contains(columnName)) {
                return true;
            }
        }
        return false;
    }

    public String bighump(String columnName) {
        return columnName.substring(0, 1).toUpperCase() + columnName.substring(1);
    }

    public static boolean isSub(String tplCategory) {
        return tplCategory != null && StringUtils.equals(GenConstants.TPL_SUB, tplCategory);
    }

    public boolean isCrud() {
        return isCrud(this.tplCategory);
    }

    public static boolean isCrud(String tplCategory) {
        return tplCategory != null && StringUtils.equals(GenConstants.TPL_CRUD, tplCategory);
    }

}
